"""Convert Megatron-Core ckpt to Qwen2-hf ckpt."""

from lingo.runner.misc.qwen2_ckpt_convertor_new import Qwen2CheckpointConvertor
from lingo.utils.constants import ModelType




misc = dict(
    type=Qwen2CheckpointConvertor,
    loader_dict=dict(
        type=ModelType.MEGATRON_CORE,
        model_path="/mnt/user/linzhixin/heatmap/model/ckpt/Qwen2.5-7B",
        params_dtype="bf16",
        hf_tokenizer="/mnt/public/open_source_model/Qwen2.5/Qwen2.5-7B",
    ),
    saver_dict=dict(
        type=ModelType.HF_STANDARD,
        model_path='/mnt/user/linzhixin/heatmap/model/hf/Qwen2.5-7B-1745',
        max_shard_size="4GB",
    ),
    mgt_path=None,
    print_checkpoint_structure=True,
)

